package com.itany.shop.util;

import com.alibaba.druid.pool.DruidDataSourceFactory;
import com.itany.shop.exception.DataAccessException;
import org.apache.commons.dbcp.BasicDataSourceFactory;

import javax.sql.DataSource;
import java.io.IOException;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.Properties;

/**
 * @author Miss Chen
 * @version 1.0
 * @date 2024年08月14日10:45
 */
public class JDBCUtil {
    private static DataSource ds;
    private static ThreadLocal<Connection> local;
    static {
        local=new ThreadLocal<>();
        Properties p=new Properties();
        try {
            p.load(JDBCUtil.class.getClassLoader().getResourceAsStream("dataSource.properties"));

            //comms-dbcp
            ds= BasicDataSourceFactory.createDataSource(p);

            // 德鲁伊
            // ds= DruidDataSourceFactory.createDataSource(p);
        } catch (IOException e) {
            e.printStackTrace();
            throw new ExceptionInInitializerError("JDBCUtil初始化失败");
        } catch (Exception e) {
            e.printStackTrace();
            throw new ExceptionInInitializerError("JDBCUtil初始化失败");
        }
    }

    /**
     * 使用数据库连接池获取一个线程单例的Connection对象
     * @return
     */
    public static Connection getConnection(){
        //从线程池中获取Connection对象、

        Connection conn=local.get();
        //如果没有获取到连接，则表示连接尚未存在

            try {
                if(null==conn){
                    //从连接池中获取出一个连接
                    conn=ds.getConnection();

                    //当拿到连接后，将连接放到线程池中
                    //当下次从线程池中获取的时候，与本次属于同一个连接
                    local.set(conn);
                }
            } catch (SQLException e) {
                e.printStackTrace();
                throw new DataAccessException("获得连接失败");
            }

            return conn;

    }

    /**
     * 该方法主要用于持久化操作之后释放
     * 释放时需要注意，Connection在业务层的事务中如果需要使用，则此时不释放
     * @param conn
     * @param ps
     * @param rs
     */
    public static void close(Connection conn, PreparedStatement ps, ResultSet rs){

            try {
                if(conn!=null){
                    conn.close();
                }
            } catch (SQLException e) {
                e.printStackTrace();
                throw new DataAccessException("释放连接失败");
            }

            try {
                    if(ps!=null){
                        ps.close();
                    }
                } catch (SQLException e) {
                    e.printStackTrace();
                    throw new DataAccessException("释放状态集失败");
            }

            try {
                if(rs!=null){
                    rs.close();
                }
            } catch (SQLException e) {
                e.printStackTrace();
                throw new DataAccessException("释放结果集失败");
            }


    }

    /**
     * 释放当前线程中相关的连接
     * 一般用于业务层事务执行完毕后释放
     */
    public static void close(){

        //从线程池中获取当前连接
        Connection conn=local.get();

        try {

            //如果获取到了连接，说明尚未释放
          if(conn!=null){

              //则需要释放资源
              conn.close();

              //需要将连接从线程池中移除
              local.remove();

          }
        } catch (SQLException e) {
            e.printStackTrace();
            throw new DataAccessException("释放连接失败");
        }

        //如果线程池中没有取到连接，说明已经释放了，无需做任何操作
    }


}
